from PIL.Image import logger
from random import randint
import re
from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.auth import login, authenticate, logout

from home.models import ArticleCategory, Article
from hopublog.models import *
from django.http import *
from django.shortcuts import *
from django.views import View
from django_redis import get_redis_connection
from libs.yuntongxun.sms import CCP
 #导入图片验证码库
# Create your views here.

# 注册视图
from libs.captcha.captcha import captcha
from urils.response_code import RETCODE


class RegisterView(View):
    # 注册页面展示
    def get(self,request):
        return render(request,'register.html')
    def post(self,request):
        mobile = request.POST.get('mobile')
        password = request.POST.get('password')
        password2 = request.POST.get('password2')
        sms_code = request.POST.get('sms_code')
        # 实现思路：
        # 1、接收用户数据
        # 2、验证数据
        # 2 - 1、参数是否齐全
        if not all([mobile,password,password2,sms_code]):
            return HttpResponseBadRequest('参数不全')
            # 2-2、手机号格式是否正确

        if not re.match(r'^1[3-9]\d{9}$', mobile):
            return HttpResponseBadRequest('手机号码格式不正确')
        # 2-3、密码是否符合格式
        if not re.match(r'^[0-9a-zA-Z]{8,20}$', password) or not re.match(r'^[0-9a-zA-Z]{8,20}$', password2):
            return HttpResponseBadRequest('密码长度、格式不正确。长度8-20，必须是数字、字母')
            # 2-4、密码和确认密码是否一致
        if password != password2:
            return HttpResponseBadRequest('两次输入的密码不一致')
        #获取redis中的短信验证码
        redis_conn = get_redis_connection()
        redis_sms_code = redis_conn.get('sms:%s' % mobile)
        if redis_sms_code.decode()!= sms_code:
            return HttpResponseBadRequest('短信验证码错误')
        # 3、保存注册信息
        user = User.objects.create_user(username=mobile,mobile=mobile,password=password)
        login(request,user)

        resp = redirect(reverse('home:index'))
        resp.set_cookie('is_login',True)
        resp.set_cookie('login_name',user.username)
        # 4、返回响应跳转到指定页面
        return resp

class ImageView(View):
    def get(self, request):
        '''
        步骤：
        1、接收前端传递过来的uuid
        2、判断uuid失分获取到
        3、通过调用captcha来生成图片验证码（返回：图片内容和图片二进制）
        4、将图片内容保存到redis中。uuid作为key，图片内容作为值，同时还需要设置一个过期时间
        5、返回图片给前端
        :param request:
        :return:
        '''

        # 1、接收前端传递过来的uuid
        uuid = request.GET.get('uuid')
        # 3、通过调用captcha来生成图片验证码
        txt, image = captcha.generate_captcha()
        # 4、将图片内容保存到redis中。uuid作为key，图片内容作为值，同时还需要设置一个过期时间
        redis_conn = get_redis_connection('default')
        # name:数据key，这里采用img前缀：uuid
        #time"300秒后过期
        #value:对应key值
        redis_conn.setex(name='img:%s' % uuid,time=300,value=txt)

        # 5、返回图片给前端
        return HttpResponse(image, content_type='image/jpeg')


class SmsCodeView(View):
    def get(self, request,):
        #1.接受参数
        mobile = request.GET.get('mobile')
        image_code = request.GET.get('image_code')
        uuid = request.GET.get('uuid')
        #2.参数校验
        #2-1.验证参数是否齐全
        if not all([mobile,image_code,uuid]):
            return JsonResponse({"code":RETCODE.IMAGECODEERR,"errmsg":'参数不齐全'})
        #2.2图片验证码的验证
        redis_conn = get_redis_connection()
        #从redis中获取已经存在的图片验证码信息
        image_code_redis = redis_conn.get('img:%s' % uuid)
        if image_code_redis is None:
            return JsonResponse({"code": RETCODE.IMAGECODEERR, "errmsg": '图片验证码不存在'})

        #删除redis中已经验证的图片验证码
        redis_conn.delete('img:%s' % uuid)
        if image_code.lower() != image_code_redis.decode().lower():
            return JsonResponse({"code": RETCODE.IMAGECODEERR, "errmsg": '图片验证码错误'})

        # 3、生成短信验证码：生成6位数验证码
        sms_code = '%06d' % randint(0, 999999)
        ccp = CCP()
        ccp.send_template_sms('17612921326', [sms_code, 5], 1)
        print(f'你的验证码是：{sms_code}')
        redis_conn.setex('sms:%s' % mobile,60,sms_code)
        return JsonResponse({"code": RETCODE.OK, "errmsg": '短信发送成功'})

class LoginView(View):
    def get(self,req):
        return render(req,'login.html')
    def post(self,req):
        '''
                实现思路：
                1、接收提交参数
                2、验证参数
                 2-1、手机号码是否符合规则
                 2-2、密码是否符合规则
                3、用户认证登录
                4、状态保持
                5、根据用户选择的是否记住登录状态进行判断
                6、设置cookie信息，为首页显示服务
                7、跳转到首页
                :param request:
                :return:
                '''
        #1.接受提交参数
        mobile = req.POST.get('mobile')
        password = req.POST.get('password')
        remeber = req.POST.get('remeber')
        #2.验证参数
        #参数是否齐全
        if not all([mobile,password]):
            return HttpResponseBadRequest('缺少必要参数')
        # 2-1.手机号码是否符合检测
        if not re.match(r'^1[3-9]\d{9}$',mobile):
            return HttpResponseBadRequest('手机号码格式不正确')
        #2-2密码格式是否符合检测
        if not re.match(r'^[0-9A-Za-z]{8,20}$',password):
            return HttpResponseBadRequest('密码格式不正确')
        #3.用户认证登录
        # 采用系统自带的认证方法进行认证
        # 如果用户名和密码正确，会返回user，否则返回None
        # 默认的认证方法是：针对于username字段进行用户名的判断。当前的判断信息是“手机号”，所以我们需要修改一下认证字段。我们需要到User模型中进行修改，等测试出现问题的时候，我们再修改
        return_user = authenticate(mobile=mobile,password=password)
        #如果账号密码验证通过，那么返回有数据的用户信息，否则返回None
        if return_user is None:
            return HttpResponseBadRequest('用户名或密码错误')
        #4.状态保持
        login(req,return_user)
        #5.根据用户选择的是否记住登录状态进行判断
        next_page = req.GET.get('next')
        if next_page:
            resp = redirect(next_page)
        else:
            resp = redirect(reverse('home:index'))
        if remeber != 'on': #没有记住用户信息
            #浏览器关闭之后过期
            # 6.设置cookie信息，为首页显示服务
            req.session.set_expiry(0)
            resp.set_cookie('is_login',True)
            resp.set_cookie('login_name',return_user.username)
        else: #记住用户信息
            #默认国企时间，其实使用了全局的过期时间，两周
            req.session.set_expiry(None)
            resp.set_cookie('is_login',True,max_age=24 * 3600 * 14)
            resp.set_cookie('login_name',return_user.username,max_age=24 * 3600 * 14)
        return resp

class LogoutView(View):
    def get(self, request):
        '''
        实现思路：
        1、清楚session数据
        2、删除cookie数据
        3、跳转到首页
        :param request:
        :return:
        '''
        # 实现思路：
        # 1、清楚session数据
        logout(request)
        # 2、删除cookie数据
        resp = redirect(reverse('home:index'))
        resp.delete_cookie('is_login')
        resp.delete_cookie('login_name')
        # 3、跳转到首页
        return resp


class ForgetPasswordView(View):
    def get(self,req):
        return render(req,'forget_password.html')
    def post(self,req):
        mobile = req.POST.get('mobile')
        password = req.POST.get('password')
        password2 = req.POST.get('password2')
        sms_code = req.POST.get('sms_code')
        #验证数据
        #判断参数是否齐全
        if not all([mobile, password]):
            return HttpResponseBadRequest('缺少必要参数')
        # 2-1.手机号码是否符合检测
        if not re.match(r'^1[3-9]\d{9}$', mobile):
            return HttpResponseBadRequest('手机号码格式不正确')
        # 2-2密码格式是否符合检测
        if not re.match(r'^[0-9A-Za-z]{8,20}$', password):
            return HttpResponseBadRequest('密码格式不正确')
        #判断确认密码是否一致
        if password != password2:
            return HttpResponseBadRequest('两次密码输入不一致')
        #判断短信验证码是否正确
        redis_conn = get_redis_connection('default')
        redis_sms_code = redis_conn.get('sms:%s' % mobile)
        if redis_sms_code is None:
            return HttpResponseBadRequest('短信验证码过期')
        try:
            if redis_sms_code.decode() != sms_code:
                return HttpResponseBadRequest('短信验证码错误')
        except Exception as e:
            logger.error(e)
            return HttpResponseBadRequest('验证码错误')
        #根据手机号码进行用户信息的查询
        return_user = User.objects.filter(mobile=mobile).first()
        #如果手机号查询出用户信息则进行用户密码的修改
        #如果手机号没有查询出用户信息，则进行新用户的创建
        if return_user is None:
            try:
                User.objects.create_user(username=mobile, mobile=mobile, password=password)
            except Exception as e:
                logger.error(e)
                return HttpResponseBadRequest('注册失败')
        else:
            try:
                # 调用系统user对象的set_password()进行修改密码，该方法会对密码进行加密
                return_user.set_password(password)
                return_user.save()
            except Exception as e:
                logger.error(e)
                return HttpResponseBadRequest('修改密码失败')
        # 6、进行页面跳转，跳转到登录页面
        resp = redirect(reverse('hopublog:login'))
        # 7、返回响应
        return resp

class UserCenterView(LoginRequiredMixin,View):

    def get(self,req):
        # if req.user.is_authenticated:
        #     return redirect(reverse('hopublog:login'))
        userinfo = req.user
        context = {
            'username' : userinfo.username,
            'mobile' : userinfo.mobile,
            'avatar' : userinfo.avatar.url if userinfo.avatar else None,
            'user_desc' : userinfo.user_desc
        }
        return render(req,'usercenter.html',context=context,)
    def post(self,req):
        #获取已经登录用户的信息
        userinfo = req.user
        #接受用户参数
        username = req.POST.get('username',userinfo.username)
        user_desc = req.POST.get('desc',userinfo.user_desc)
        avatar = req.FILES.get('avatar')
        try:
            userinfo.username = username
            userinfo.user_desc = user_desc
            if avatar:
                userinfo.avatar = avatar
            userinfo.save()
        except Exception as e:
            logger.error(e)
            return HttpResponseBadRequest('修改用户信息失败')
        resp = redirect(reverse('home:index'))
        resp.set_cookie('login_name',userinfo.username)
        return resp
class WriteBlogView(View):
    def get(self,req):
        contxt=ArticleCategory.objects.all()
        return render(req,'writblog.html',{'list':contxt})
# 只有登录的用户才能写博客，所以需要对匿名身份验证
    def post(self, request):
        '''
        实现思路：
        1、接收数据
        2、验证数据
        3、数据入库
        4、跳转到指定页面
        :param request:
        :return:
        '''
        # 1、接收数据
        avatar = request.FILES.get('avatar')
        title = request.POST.get('title')
        category_id = request.POST.get('category')
        tags = request.POST.get('tags')
        sumary = request.POST.get('sumary')
        content = request.POST.get('content')
        user = request.user
        print(avatar,title,category_id,tags,sumary,content)
        # 2、验证数据
        # 2-1、参数齐全验证
        if not all([avatar, title, category_id, sumary, content]):
            return HttpResponseBadRequest('参数不齐全')
        # 2-2、判断分类id
        try:
            category = ArticleCategory.objects.filter(pk=category_id).first()
        except ArticleCategory.DoesNotExist:
            return HttpResponseBadRequest('没有该分类信息')
        # 3、数据入库
        try:
            Article.objects.create(
                author=user,
                avatar=avatar,
                category=category,
                tags=tags,
                sumary=sumary,
                content=content,
                title=title,
            )

        except Exception as e:
            logger.error(e)
            return HttpResponseBadRequest('发布失败，请稍后重试')
        # 4、跳转到指定页面
        return redirect(reverse('home:index'))
















